using key to alter sorting order in sorted() in python